#adapt_alg="PLL"
adapt_alg="PLL_imbalance"
dataset="DomainNet"
domain="clipart"
test_envs=0
network="resnet18"

lr=1e-5
margin_select=0.2
margin_end=0.1
margin_descend=$(echo "scale=4; (${margin_select} - ${margin_end}) / 382" | bc)
temp_select=1
temp_loss=1
buffer_size=32
update_param='all'

seed=0
seed_array=(0)
gpu_id=2

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python unsupervise_adapt.py --dataset ${dataset} --data_dir /data/huyihao/projects/data/${dataset} --adapt_alg ${adapt_alg} --pretrain_dir train_output/${network}/${dataset}/${domain}/${seed}/model.pkl --lr ${lr} --net ${network} --test_envs ${test_envs} --seed ${seed} --output tta_output/${network}/${dataset}/${domain}/${adapt_alg} --margin_select ${margin_select} --margin_end ${margin_end} --margin_descend ${margin_descend} --temp_select ${temp_select} --temp_loss ${temp_loss} --buffer_size ${buffer_size} --update_param ${update_param}
done